home *** CD-ROM | disk | FTP | other *** search
/ Komputer for Alle 2004 #2 / K-CD-2-2004.ISO / OpenOffice Sv / f_0397 / python-core-2.2.2 / lib / test / test_mhlib.py < prev    next >
Encoding:
Python Source  |  2003-07-18  |  10.3 KB  |  341 lines
  1. """
  2.    Tests for the mhlib module
  3.    Nick Mathewson
  4. """
  5.  
  6. ### BUG: This suite doesn't currently test the mime functionality of
  7. ###      mhlib.  It should.
  8.  
  9. import unittest
  10. from test_support import run_unittest, TESTFN, TestSkipped
  11. import os, StringIO
  12. import sys
  13. import mhlib
  14.  
  15. if sys.platform.startswith("win") or sys.platform=="riscos":
  16.     raise TestSkipped("test_mhlib skipped on %s -- "%sys.platform +
  17.                       "too many Unix assumptions")
  18.  
  19. _mhroot = TESTFN+"_MH"
  20. _mhpath = os.path.join(_mhroot, "MH")
  21. _mhprofile = os.path.join(_mhroot, ".mh_profile")
  22.  
  23. def normF(f):
  24.     return os.path.join(*f.split('/'))
  25.  
  26. def writeFile(fname, contents):
  27.     dir = os.path.split(fname)[0]
  28.     if dir and not os.path.exists(dir):
  29.         mkdirs(dir)
  30.     f = open(fname, 'w')
  31.     f.write(contents)
  32.     f.close()
  33.  
  34. def readFile(fname):
  35.     f = open(fname)
  36.     r = f.read()
  37.     f.close()
  38.     return r
  39.  
  40. def writeProfile(dict):
  41.     contents = [ "%s: %s\n" % (k, v) for k, v in dict.iteritems() ]
  42.     writeFile(_mhprofile, "".join(contents))
  43.  
  44. def writeContext(folder):
  45.     folder = normF(folder)
  46.     writeFile(os.path.join(_mhpath, "context"),
  47.               "Current-Folder: %s\n" % folder)
  48.  
  49. def writeCurMessage(folder, cur):
  50.     folder = normF(folder)
  51.     writeFile(os.path.join(_mhpath, folder, ".mh_sequences"),
  52.               "cur: %s\n"%cur)
  53.  
  54. def writeMessage(folder, n, headers, body):
  55.     folder = normF(folder)
  56.     headers = "".join([ "%s: %s\n" % (k, v) for k, v in headers.iteritems() ])
  57.     contents = "%s\n%s\n" % (headers,body)
  58.     mkdirs(os.path.join(_mhpath, folder))
  59.     writeFile(os.path.join(_mhpath, folder, str(n)), contents)
  60.  
  61. def getMH():
  62.     return mhlib.MH(os.path.abspath(_mhpath), _mhprofile)
  63.  
  64. def sortLines(s):
  65.     lines = s.split("\n")
  66.     lines = [ line.strip() for line in lines if len(line) >= 2 ]
  67.     lines.sort()
  68.     return lines
  69.  
  70. # These next 2 functions are copied from test_glob.py.
  71. def mkdirs(fname):
  72.     if os.path.exists(fname) or fname == '':
  73.         return
  74.     base, file = os.path.split(fname)
  75.     mkdirs(base)
  76.     os.mkdir(fname)
  77.  
  78. def deltree(fname):
  79.     if not os.path.exists(fname):
  80.         return
  81.     for f in os.listdir(fname):
  82.         fullname = os.path.join(fname, f)
  83.         if os.path.isdir(fullname):
  84.             deltree(fullname)
  85.         else:
  86.             try:
  87.                 os.unlink(fullname)
  88.             except:
  89.                 pass
  90.     try:
  91.         os.rmdir(fname)
  92.     except:
  93.         pass
  94.  
  95. class MhlibTests(unittest.TestCase):
  96.     def setUp(self):
  97.         deltree(_mhroot)
  98.         mkdirs(_mhpath)
  99.         writeProfile({'Path' : os.path.abspath(_mhpath),
  100.                       'Editor': 'emacs',
  101.                       'ignored-attribute': 'camping holiday'})
  102.         # Note: These headers aren't really conformant to RFC822, but
  103.         #  mhlib shouldn't care about that.
  104.  
  105.         # An inbox with a couple of messages.
  106.         writeMessage('inbox', 1,
  107.                      {'From': 'Mrs. Premise',
  108.                       'To': 'Mrs. Conclusion',
  109.                       'Date': '18 July 2001'}, "Hullo, Mrs. Conclusion!\n")
  110.         writeMessage('inbox', 2,
  111.                      {'From': 'Mrs. Conclusion',
  112.                       'To': 'Mrs. Premise',
  113.                       'Date': '29 July 2001'}, "Hullo, Mrs. Premise!\n")
  114.  
  115.         # A folder with many messages
  116.         for i in range(5, 101)+range(101, 201, 2):
  117.             writeMessage('wide', i,
  118.                          {'From': 'nowhere', 'Subject': 'message #%s' % i},
  119.                          "This is message number %s\n" % i)
  120.  
  121.         # A deeply nested folder
  122.         def deep(folder, n):
  123.             writeMessage(folder, n,
  124.                          {'Subject': 'Message %s/%s' % (folder, n) },
  125.                          "This is message number %s in %s\n" % (n, folder) )
  126.         deep('deep/f1', 1)
  127.         deep('deep/f1', 2)
  128.         deep('deep/f1', 3)
  129.         deep('deep/f2', 4)
  130.         deep('deep/f2', 6)
  131.         deep('deep', 3)
  132.         deep('deep/f2/f3', 1)
  133.         deep('deep/f2/f3', 2)
  134.  
  135.     def tearDown(self):
  136.         deltree(_mhroot)
  137.  
  138.     def test_basic(self):
  139.         writeContext('inbox')
  140.         writeCurMessage('inbox', 2)
  141.         mh = getMH()
  142.  
  143.         eq = self.assertEquals
  144.         eq(mh.getprofile('Editor'), 'emacs')
  145.         eq(mh.getprofile('not-set'), None)
  146.         eq(mh.getpath(), os.path.abspath(_mhpath))
  147.         eq(mh.getcontext(), 'inbox')
  148.  
  149.         mh.setcontext('wide')
  150.         eq(mh.getcontext(), 'wide')
  151.         eq(readFile(os.path.join(_mhpath, 'context')),
  152.            "Current-Folder: wide\n")
  153.  
  154.         mh.setcontext('inbox')
  155.  
  156.         inbox = mh.openfolder('inbox')
  157.         eq(inbox.getfullname(),
  158.            os.path.join(os.path.abspath(_mhpath), 'inbox'))
  159.         eq(inbox.getsequencesfilename(),
  160.            os.path.join(os.path.abspath(_mhpath), 'inbox', '.mh_sequences'))
  161.         eq(inbox.getmessagefilename(1),
  162.            os.path.join(os.path.abspath(_mhpath), 'inbox', '1'))
  163.  
  164.     def test_listfolders(self):
  165.         mh = getMH()
  166.         eq = self.assertEquals
  167.  
  168.         folders = mh.listfolders()
  169.         folders.sort()
  170.         eq(folders, ['deep', 'inbox', 'wide'])
  171.  
  172.         folders = mh.listallfolders()
  173.         folders.sort()
  174.         tfolders = map(normF, ['deep', 'deep/f1', 'deep/f2', 'deep/f2/f3',
  175.                                 'inbox', 'wide'])
  176.         tfolders.sort()
  177.         eq(folders, tfolders)
  178.  
  179.         folders = mh.listsubfolders('deep')
  180.         folders.sort()
  181.         eq(folders, map(normF, ['deep/f1', 'deep/f2']))
  182.  
  183.         folders = mh.listallsubfolders('deep')
  184.         folders.sort()
  185.         eq(folders, map(normF, ['deep/f1', 'deep/f2', 'deep/f2/f3']))
  186.         eq(mh.listsubfolders(normF('deep/f2')), [normF('deep/f2/f3')])
  187.  
  188.         eq(mh.listsubfolders('inbox'), [])
  189.         eq(mh.listallsubfolders('inbox'), [])
  190.  
  191.     def test_sequence(self):
  192.         mh = getMH()
  193.         eq = self.assertEquals
  194.         writeCurMessage('wide', 55)
  195.  
  196.         f = mh.openfolder('wide')
  197.         all = f.listmessages()
  198.         eq(all, range(5, 101)+range(101, 201, 2))
  199.         eq(f.getcurrent(), 55)
  200.         f.setcurrent(99)
  201.         eq(readFile(os.path.join(_mhpath, 'wide', '.mh_sequences')),
  202.            'cur: 99\n')
  203.  
  204.         def seqeq(seq, val):
  205.             eq(f.parsesequence(seq), val)
  206.  
  207.         seqeq('5-55', range(5, 56))
  208.         seqeq('90-108', range(90, 101)+range(101, 109, 2))
  209.         seqeq('90-108', range(90, 101)+range(101, 109, 2))
  210.  
  211.         seqeq('10:10', range(10, 20))
  212.         seqeq('10:+10', range(10, 20))
  213.         seqeq('101:10', range(101, 121, 2))
  214.  
  215.         seqeq('cur', [99])
  216.         seqeq('.', [99])
  217.         seqeq('prev', [98])
  218.         seqeq('next', [100])
  219.         seqeq('cur:-3', [97, 98, 99])
  220.         seqeq('first-cur', range(5, 100))
  221.         seqeq('150-last', range(151, 201, 2))
  222.         seqeq('prev-next', [98, 99, 100])
  223.  
  224.         lowprimes = [5, 7, 11, 13, 17, 19, 23, 29]
  225.         lowcompos = [x for x in range(5, 31) if not x in lowprimes ]
  226.         f.putsequences({'cur': [5],
  227.                         'lowprime': lowprimes,
  228.                         'lowcompos': lowcompos})
  229.         seqs = readFile(os.path.join(_mhpath, 'wide', '.mh_sequences'))
  230.         seqs = sortLines(seqs)
  231.         eq(seqs, ["cur: 5",
  232.                   "lowcompos: 6 8-10 12 14-16 18 20-22 24-28 30",
  233.                   "lowprime: 5 7 11 13 17 19 23 29"])
  234.  
  235.         seqeq('lowprime', lowprimes)
  236.         seqeq('lowprime:1', [5])
  237.         seqeq('lowprime:2', [5, 7])
  238.         seqeq('lowprime:-2', [23, 29])
  239.  
  240.         ## Not supported
  241.         #seqeq('lowprime:first', [5])
  242.         #seqeq('lowprime:last', [29])
  243.         #seqeq('lowprime:prev', [29])
  244.         #seqeq('lowprime:next', [29])
  245.  
  246.     def test_modify(self):
  247.         mh = getMH()
  248.         eq = self.assertEquals
  249.  
  250.         mh.makefolder("dummy1")
  251.         self.assert_("dummy1" in mh.listfolders())
  252.         path = os.path.join(_mhpath, "dummy1")
  253.         self.assert_(os.path.exists(path))
  254.  
  255.         f = mh.openfolder('dummy1')
  256.         def create(n):
  257.             msg = "From: foo\nSubject: %s\n\nDummy Message %s\n" % (n,n)
  258.             f.createmessage(n, StringIO.StringIO(msg))
  259.  
  260.         create(7)
  261.         create(8)
  262.         create(9)
  263.  
  264.         eq(readFile(f.getmessagefilename(9)),
  265.            "From: foo\nSubject: 9\n\nDummy Message 9\n")
  266.  
  267.         eq(f.listmessages(), [7, 8, 9])
  268.         files = os.listdir(path)
  269.         files.sort()
  270.         eq(files, ['7', '8', '9'])
  271.  
  272.         f.removemessages(['7', '8'])
  273.         files = os.listdir(path)
  274.         files.sort()
  275.         eq(files, [',7', ',8', '9'])
  276.         eq(f.listmessages(), [9])
  277.         create(10)
  278.         create(11)
  279.         create(12)
  280.  
  281.         mh.makefolder("dummy2")
  282.         f2 = mh.openfolder("dummy2")
  283.         eq(f2.listmessages(), [])
  284.         f.movemessage(10, f2, 3)
  285.         f.movemessage(11, f2, 5)
  286.         eq(f.listmessages(), [9, 12])
  287.         eq(f2.listmessages(), [3, 5])
  288.         eq(readFile(f2.getmessagefilename(3)),
  289.            "From: foo\nSubject: 10\n\nDummy Message 10\n")
  290.  
  291.         f.copymessage(9, f2, 4)
  292.         eq(f.listmessages(), [9, 12])
  293.         eq(readFile(f2.getmessagefilename(4)),
  294.            "From: foo\nSubject: 9\n\nDummy Message 9\n")
  295.  
  296.         f.refilemessages([9, 12], f2)
  297.         eq(f.listmessages(), [])
  298.         eq(f2.listmessages(), [3, 4, 5, 6, 7])
  299.         eq(readFile(f2.getmessagefilename(7)),
  300.            "From: foo\nSubject: 12\n\nDummy Message 12\n")
  301.         # XXX This should check that _copysequences does the right thing.
  302.  
  303.         mh.deletefolder('dummy1')
  304.         mh.deletefolder('dummy2')
  305.         self.assert_('dummy1' not in mh.listfolders())
  306.         self.assert_(not os.path.exists(path))
  307.  
  308.     def test_read(self):
  309.         mh = getMH()
  310.         eq = self.assertEquals
  311.  
  312.         f = mh.openfolder('inbox')
  313.         msg = f.openmessage(1)
  314.         # Check some basic stuff from rfc822
  315.         eq(msg.getheader('From'), "Mrs. Premise")
  316.         eq(msg.getheader('To'), "Mrs. Conclusion")
  317.  
  318.         # Okay, we have the right message.  Let's check the stuff from
  319.         # mhlib.
  320.         lines = sortLines(msg.getheadertext())
  321.         eq(lines, ["Date: 18 July 2001",
  322.                    "From: Mrs. Premise",
  323.                    "To: Mrs. Conclusion"])
  324.         lines = sortLines(msg.getheadertext(lambda h: len(h)==4))
  325.         eq(lines, ["Date: 18 July 2001",
  326.                    "From: Mrs. Premise"])
  327.         eq(msg.getbodytext(), "Hullo, Mrs. Conclusion!\n\n")
  328.         eq(msg.getbodytext(0), "Hullo, Mrs. Conclusion!\n\n")
  329.  
  330.         # XXXX there should be a better way to reclaim the file handle
  331.         msg.fp.close()
  332.         del msg
  333.  
  334.  
  335. def test_main():
  336.     run_unittest(MhlibTests)
  337.  
  338.  
  339. if __name__ == "__main__":
  340.     test_main()
  341.